package com.ljk.util.pdf.data;

import com.itextpdf.text.*;
import com.itextpdf.text.pdf.*;
import lombok.Getter;
import lombok.Setter;

import java.io.IOException;

@Getter
@Setter
public class PageNumberEvent extends PdfPageEventHelper {

    /**
     * 文档字体大小，页脚页眉最好和文本大小一致
     */
    private int fontSize;

    /**
     * 文档页面大小，最好前面传入，否则默认为A4纸张
     */
    private Rectangle pageSize;

    // 模板
    private com.itextpdf.text.pdf.PdfTemplate total;

    /** 基础字体对象 */
    private BaseFont baseFont;

    /** 利用基础字体生成的字体对象，一般用于生成中文文字 */
    private Font font;


    public BaseFont getBaseFont() {
        try {
            if (baseFont == null) {
                baseFont = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", false);
            }
        } catch (DocumentException | IOException e) {
            throw new RuntimeException("字体获取失败");
        }
        return baseFont;
    }

    public Font getFont() {
        if (font == null) {
            font = new Font(getBaseFont(), fontSize, Font.NORMAL);// 数据体字体
        }
        return font;
    }

    public PageNumberEvent(int fontSize, Rectangle pageSize) {
        this.fontSize = fontSize;
        this.pageSize = pageSize;
    }


    /**
     * 文档打开时创建模板
     *
     * @see PdfPageEventHelper#onOpenDocument(PdfWriter,
     * Document)
     */
    @Override
    public void onOpenDocument(PdfWriter writer, Document document) {
        total = writer.getDirectContent().createTemplate(60, 60);// 共 页 的矩形的长宽高
    }

    /**
     * 关闭每页的时候，写入页眉，写入'第几页共'这几个字。
     *
     * @see PdfPageEventHelper#onEndPage(PdfWriter,
     * Document)
     */
    @Override
    public void onEndPage(PdfWriter writer, Document document) {
        this.addPage(writer, document);
        //加水印
//        this.addWatermark(writer);
    }

    /**
     * 关闭文档时，替换模板，完成整个页眉页脚组件
     *
     * @see PdfPageEventHelper#onCloseDocument(PdfWriter,
     * Document)
     */
    @Override
    public void onCloseDocument(PdfWriter writer, Document document) {
        // 7.最后一步了，就是关闭文档的时候，将模板替换成实际的 Y 值,至此，page x of y 制作完毕，完美兼容各种文档size。
        total.beginText();
        total.setFontAndSize(getBaseFont(), fontSize);// 生成的模版的字体、颜色
        String foot2 = " " + (writer.getPageNumber()) + " 页"; //页脚内容拼接  如  第1页/共2页
        total.showText(foot2);// 模版显示的内容
        total.endText();
        total.closePath();
    }

    //加分页
    public void addPage(PdfWriter writer, Document document) {
        //设置分页页眉页脚字体


//         1.写入页眉
//        ColumnText.showTextAligned(writer.getDirectContent(),
//                Element.ALIGN_LEFT, new Phrase("页眉", getFont()),
//                document.left(), document.top() - 20, 0);


        // 2.写入前半部分的 第 X页/共
        int pageS = writer.getPageNumber();
        String foot1 = "第 " + pageS + " 页 /共";
        Phrase footer = new Phrase(foot1, getFont());

        // 3.计算前半部分的foot1的长度，后面好定位最后一部分的'Y页'这俩字的x轴坐标，字体长度也要计算进去 = len
        float len = getBaseFont().getWidthPoint(foot1, fontSize);

        // 4.拿到当前的PdfContentByte
        PdfContentByte cb = writer.getDirectContent();


        float centerX = (document.rightMargin() + document.right() + document.leftMargin() - document.left()) / 2.0F;
        float centerY =  17;
        // 5.写入页脚1，x轴就是(右margin+左margin + right() -left()- len)/2.0F
        // 再给偏移20F适合人类视觉感受，否则肉眼看上去就太偏左了
        // ,y轴就是底边界-20,否则就贴边重叠到数据体里了就不是页脚了；注意Y轴是从下往上累加的，最上方的Top值是大于Bottom好几百开外的。
        ColumnText.showTextAligned(cb, Element.ALIGN_LEFT, footer, centerX - len, centerY, 0);
        // 6.写入页脚2的模板（就是页脚的Y页这俩字）添加到文档中，计算模板的和Y轴,X=(右边界-左边界 - 前半部分的len值)/2.0F +
        // len ， y 轴和之前的保持一致，底边界-20
        cb.addTemplate(total, centerX, centerY); // 调节模版显示的位置

    }


    //加水印
    public void addWatermark(PdfWriter writer) {
        // 水印图片
//        Image image;
//        try {
//            image = Image.getInstance("./web/images/001.jpg");
//            PdfContentByte content = writer.getDirectContentUnder();
//            content.beginText();
//            // 开始写入水印
//            for(int k=0;k<5;k++){
//                for (int j = 0; j <4; j++) {
//                    image.setAbsolutePosition(150*j,170*k);
//                    content.addImage(image);
//                }
//            }
//            content.endText();
//        } catch (IOException | DocumentException e) {
//            // TODO Auto-generated catch block
//            e.printStackTrace();
//        }
/*        BaseFont font = null;
        try {
            font = BaseFont.createFont("STSong-Light","UniGB-UCS2-H", BaseFont.NOT_EMBEDDED);
        } catch (DocumentException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        PdfGState gs = new PdfGState();
        //添加透明度
        gs.setFillOpacity(0.4f);
        PdfContentByte content = writer.getDirectContentUnder();
        content.beginText();
        //水印颜色
        content.setColorFill(BaseColor.DARK_GRAY);
        content.setGState(gs);
        //水印字体样式和大小
        content.setFontAndSize(font, 35);
        //插入水印  循环每页插入的条数
        for (int j = 0; j < 3; j++) {
            content.showTextAligned(Element.ALIGN_CENTER, "锦鲤飞上天测试水印", 300, 200 * (j + 1), 30);
        }
        content.endText();*/
    }

}
